package com.easy.util;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

public class ExpiryMap {
    // 存储实际数据
    private Map<String, Object> map = new HashMap<>();
    // 存储过期时间戳
    private Map<String, Long> expiryMap = new HashMap<>();
    // 定时清理线程池
    private ScheduledExecutorService service = Executors.newSingleThreadScheduledExecutor();
    // 默认过期时间（6秒）
    private long expiryTime = 1000 * 6;

    public ExpiryMap() {
        // 定时清理过期数据
        service.scheduleAtFixedRate(this::clearExpired, expiryTime, expiryTime, TimeUnit.MILLISECONDS);
    }

    public ExpiryMap(long expiryTime) {
        this.expiryTime = expiryTime;
        service.scheduleAtFixedRate(this::clearExpired, expiryTime, expiryTime, TimeUnit.MILLISECONDS);
    }

    // 添加数据（默认过期时间）
    public void put(String key, Object value) {
        map.put(key, value);
        expiryMap.put(key, System.currentTimeMillis() + expiryTime);
    }

    // 添加数据（自定义过期时间）
    public void put(String key, Object value, long expiryTime) {
        map.put(key, value);
        expiryMap.put(key, System.currentTimeMillis() + expiryTime);
    }

    // 获取数据（修复空指针关键处）
    public Object get(String key) {
        // 1. 先判断key是否存在，不存在直接返回null
        if (!expiryMap.containsKey(key)) {
            return null;
        }

        // 2. 存在则检查是否过期
        Long expireTime = expiryMap.get(key);
        if (expiryMap.get(key)!=null&&System.currentTimeMillis() > expireTime) {
            // 过期则移除
            map.remove(key);
            expiryMap.remove(key);
            return null;
        }

        return map.get(key);
    }

    // 移除数据
    public void remove(String key) {
        map.remove(key);
        expiryMap.remove(key);
    }

    // 清理过期数据（使用迭代器避免并发修改异常）
    public void clearExpired() {
        Iterator<String> iterator = expiryMap.keySet().iterator();
        while (iterator.hasNext()) {
            String key = iterator.next();
            if (System.currentTimeMillis() > expiryMap.get(key)) {
                map.remove(key);
                iterator.remove(); // 安全移除
            }
        }
    }

    // 关闭线程池
    public void shutdown() {
        service.shutdown();
    }
}
